package com.sunrise.generico.dataModel;  
  
import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import com.sunrise.generico.entidade.Produto;

public class ProdutoDataModel extends ListDataModel<Produto> implements SelectableDataModel<Produto> {    
  
    public ProdutoDataModel() {  
    }  
  
    public ProdutoDataModel(List<Produto> data) {  
        super(data);  
    }  
      
    @Override  
    public Produto getRowData(String rowKey) {  
        @SuppressWarnings("unchecked")
		List<Produto> produtos = (List<Produto>) getWrappedData();  
          
        for(Produto produto : produtos) {  
            if(produto.getId().toString().equals(rowKey))  
                return produto;  
        }  
          
        return null;  
    }  
  
    @Override  
    public Object getRowKey(Produto produto) {
    	if(produto != null){
    		return produto.getId();  
    	}
    	return null;
    }
} 